how to draw a square in python 3